import random

import pandas as pd
import openai
import os
from time import sleep

# 设置 OpenAI API 密钥
openai.api_key = \
    ''  # 用你自己的密钥替换


# 读取 Excel 文件
def read_excel(file_path):
    return pd.read_excel(file_path)


# 查询手机设备的上市时间和停售时间
def get_phone_release_info(phone_name):

    prompt = f"查询 {phone_name} 的上市时间和停售时间"

    try:
        # 使用新的 openai.completions.create 方法
        response = openai.completions.create(
            model="gpt-3.5-turbo",  # 使用适合的模型（例如 gpt-3.5-turbo）
            prompt=prompt,
            max_tokens=100
        )
        return response['choices'][0]['text'].strip()
    except Exception as e:
        return f"Error: {str(e)}"


# 将查询结果写回 Excel 文件
def write_to_excel(df, output_file):
    df.to_excel(output_file, index=False)


# 主函数
def main(input_file, output_file):
    # 读取 Excel 文件
    df = read_excel(input_file)

    # 确保 Excel 文件中有设备名称列
    if '设备型号' not in df.columns:
        print("Excel 文件没有 '设备型号' 列")
        return

    # 查询每个手机设备的信息
    release_dates = []
    for index, row in df.iterrows():
        phone_name = row['设备型号']
        release_info = get_phone_release_info(phone_name)
        release_dates.append(release_info)

    # 将查询结果写入新列
    df['上市与停售时间'] = release_dates

    # 写回到新的 Excel 文件
    write_to_excel(df, output_file)
    print(f"结果已写入 {output_file}")


# 示例：输入文件路径和输出文件路径
input_file = 'D:\\workNote\\钉钉\\设备型号副本.xlsx'  # 这里填上你的输入文件路径
output_file = '设备查询结果.xlsx'  # 输出的结果文件

# 调用主函数
main(input_file, output_file)
